package com.sh4.util.wechatUtils;

import java.io.InputStream;
import java.io.Writer;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;

import javax.servlet.http.HttpServletRequest;

import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.DocumentHelper;
import org.dom4j.Element;
import org.dom4j.Namespace;
import org.dom4j.QName;
import org.dom4j.io.SAXReader;

import com.sh4.wxservice.message.resp.Article;
import com.sh4.wxservice.message.resp.MusicMessage;
import com.sh4.wxservice.message.resp.NewsMessage;
import com.sh4.wxservice.message.resp.TextMessage;
import com.thoughtworks.xstream.XStream;
import com.thoughtworks.xstream.core.util.QuickWriter;
import com.thoughtworks.xstream.io.HierarchicalStreamWriter;
import com.thoughtworks.xstream.io.xml.PrettyPrintWriter;
import com.thoughtworks.xstream.io.xml.XppDriver;

public class MessageUtil {
    /**
     * 返回消息类型：文本
     */
    public static final String RESP_MESSAGE_TYPE_TEXT = "text";

    /**
     * 返回消息类型：音乐
     */
    public static final String RESP_MESSAGE_TYPE_MUSIC = "music";

    /**
     * 返回消息类型：图文
     */
    public static final String RESP_MESSAGE_TYPE_NEWS = "news";

    /**
     * 请求消息类型：文本
     */
    public static final String REQ_MESSAGE_TYPE_TEXT = "text";

    /**
     * 请求消息类型：图片
     */
    public static final String REQ_MESSAGE_TYPE_IMAGE = "image";

    /**
     * 请求消息类型：链接
     */
    public static final String REQ_MESSAGE_TYPE_LINK = "link";

    /**
     * 请求消息类型：地理位置
     */
    public static final String REQ_MESSAGE_TYPE_LOCATION = "location";

    /**
     * 请求消息类型：音频
     */
    public static final String REQ_MESSAGE_TYPE_VOICE = "voice";

    /**
     * 请求消息类型：推送
     */
    public static final String REQ_MESSAGE_TYPE_EVENT = "event";

    /**
     * 事件类型：SCAN (用户已关注时的事件)
     */
    public static final String EVENT_TYPE_ALREADY_SUBSCRIBE = "SCAN";

    /**
     * 事件类型：VIEW (点击菜单跳转链接时的事件推送)
     */
    public static final String EVENT_TYPE_REDIRECT = "VIEW";
    /**
     * 事件类型：subscribe(订阅)
     */
    public static final String EVENT_TYPE_SUBSCRIBE = "subscribe";

    /**
     * 事件类型：unsubscribe(取消订阅)
     */
    public static final String EVENT_TYPE_UNSUBSCRIBE = "unsubscribe";

    /**
     * 事件类型：CLICK(自定义菜单点击事件)
     */
    public static final String EVENT_TYPE_CLICK = "CLICK";
    
    /**
     * 事件类型：CLICK(发送位置)
     */
    public static final String EVENT_TYPE_LOCATION = "rselfmenu_2_0";
    


    /**
     * 解析微信发来的请求（XML）
     *
     * @param request
     * @return
     * @throws Exception
     */
    @SuppressWarnings("unchecked")
    public static Map<String, Object> parseXml(HttpServletRequest request) throws Exception {

        // 将解析结果存储在HashMap中
        Map<String, Object> map = new HashMap<String, Object>();

        // 从request中取得输入流
        InputStream inputStream = request.getInputStream();
        // 读取输入流
        SAXReader reader = new SAXReader();
        Document document = reader.read(inputStream);
        // 得到xml根元素
        Element root = document.getRootElement();
        // 得到根元素的所有子节点
        List<Element> elementList = root.elements();

        // 遍历所有子节点
        for (Element e : elementList){
        	if(e.elements().size()!=0){
        		Map<String, Object> mapa = new HashMap<>();
            	 element2Map(mapa, e);
            	 map.put(e.getName(), mapa);
        	}else{
        		 map.put(e.getName(), e.getText());
        	}
        }
        // 释放资源
        inputStream.close();
        inputStream = null;
        return map;
    }
    /** 
     * 使用递归调用将多层级xml转为map 
     * @param map 
     * @param rootElement 
     */  
    @SuppressWarnings({ "unchecked", "rawtypes" })
	public static void element2Map(Map<String, Object> map, Element rootElement) {  
            
          //获得当前节点的子节点  
          List<Element> elements = rootElement.elements();  
          if (elements.size() == 0) {  
                //没有子节点说明当前节点是叶子节点，直接取值  
                map.put(rootElement.getName(),rootElement.getText());  
          }else if (elements.size() == 1) {  
                //只有一个子节点说明不用考虑list的情况，继续递归  
                Map<String,Object> tempMap = new HashMap<String,Object>();  
                element2Map(tempMap,elements.get(0));  
                map.put(rootElement.getName(),tempMap);  
          }else {  
                //多个子节点的话就要考虑list的情况了，特别是当多个子节点有名称相同的字段时  
                Map<String,Object> tempMap = new HashMap<String,Object>();  
                for (Element element : elements) {  
                      tempMap.put(element.getName(),null);  
                }  
                Set<String> keySet = tempMap.keySet();  
                for (String string : keySet) {  
                      Namespace namespace = elements.get(0).getNamespace();  
                      List<Element> sameElements = rootElement.elements(new QName(string,namespace));  
                      //如果同名的数目大于1则表示要构建list  
                      if (sameElements.size() > 1) {  
                            List<Map> list = new ArrayList<Map>();  
                            for(Element element : sameElements){  
                                  Map<String,Object> sameTempMap = new HashMap<String,Object>();  
                                  element2Map(sameTempMap,element);  
                                  list.add(sameTempMap);  
                            }  
                            map.put(string,list);  
                      }else {  
                            //同名的数量不大于1直接递归  
                            Map<String,Object> sameTempMap = new HashMap<String,Object>();  
                            element2Map(sameTempMap,sameElements.get(0));  
//                            map.put(string,sameElements.get(0));  
                            String key = getKeyOrNull(sameTempMap); 
                            map.put(key, sameTempMap.get(key));
                      }  
                }  
          }  
    }  
    /**
     * 解析微商城 同步微信 发来的请求（XML）
     *
     * @param request
     * @return
     * @throws Exception
     */
    @SuppressWarnings("unchecked")
	public static Map<String, String> parseMallXml(HttpServletRequest request) throws Exception {

        // 将解析结果存储在HashMap中
        Map<String, String> map = new HashMap<String, String>();
        String xmlStr = request.getParameter("xml");
        System.out.println(xmlStr+"   =====end");
        Document document = MessageUtil.StringToXml(xmlStr);
         document.asXML();
        // 得到xml根元素
        Element root = document.getRootElement();
        // 得到根元素的所有子节点
        List<Element> elementList = root.elements();

        // 遍历所有子节点
        for (Element e : elementList)
            map.put(e.getName(), e.getText());

        return map;
    }

    /**
     * String 转换成 xml
     * @throws DocumentException
     */
    public static Document StringToXml(String xmlStr) throws DocumentException {
        Document document = DocumentHelper.parseText(xmlStr);
        return document;
    }

    /**
     * 文本消息对象转换成xml
     *
     * @param textMessage
     *            文本消息对象
     * @return xml
     */
    public static String textMessageToXml(TextMessage textMessage) {
        xstream.alias("xml", textMessage.getClass());
        return xstream.toXML(textMessage);
    }

    /**
     * 音乐消息对象转换成xml
     *
     * @param musicMessage
     *            音乐消息对象
     * @return xml
     */
    public static String musicMessageToXml(MusicMessage musicMessage) {
        xstream.alias("xml", musicMessage.getClass());
        return xstream.toXML(musicMessage);
    }

    /**
     * 图文消息对象转换成xml
     *
     * @param newsMessage
     *            图文消息对象
     * @return xml
     */
    public static String newsMessageToXml(NewsMessage newsMessage) {
        xstream.alias("xml", newsMessage.getClass());
        xstream.alias("item", new Article().getClass());
        return xstream.toXML(newsMessage);
    }

    /**
     * 扩展xstream，使其支持CDATA块
     *
     * @date 2013-05-19
     */
    private static XStream xstream = new XStream(new XppDriver() {
        public HierarchicalStreamWriter createWriter(Writer out) {
            return new PrettyPrintWriter(out) {
                // 对所有xml节点的转换都增加CDATA标记
                boolean cdata = true;

                public void startNode(String name, @SuppressWarnings("rawtypes") Class clazz) {
                    super.startNode(name, clazz);
                }

                protected void writeText(QuickWriter writer, String text) {
                    if (cdata) {
                        writer.write("<![CDATA[");
                        writer.write(text);
                        writer.write("]]>");
                    } else {
                        writer.write(text);
                    }
                }
            };
        }
    });
    /** 
     * 获取map中第一个key值 
     * 
     * @param map 数据源 
     * @return  
     */  
	private static String getKeyOrNull(Map<String, Object> map) {  
        String obj = null;  
        for (Entry<String, Object> entry : map.entrySet()) {  
            obj = entry.getKey();  
            if (obj != null) {  
                break;  
            }  
        }  
        return  obj;  
    }  
      
      
    /** 
     * 获取map中第一个数据值 
     * 
     * @param map 数据源 
     * @return  
     */  
    @SuppressWarnings("unused")
	private static Object getFirstOrNull(Map<String, Object> map) {  
        Object obj = null;  
        for (Entry<String, Object> entry : map.entrySet()) {  
            obj = entry.getValue();  
            if (obj != null) {  
                break;  
            }  
        }  
        return  obj;  
    }  
}

